// Copyright (c) 2008 Michael Buckley

// Permission is hereby granted, free of charge, to any person
// obtaining a copy of this software and associated documentation
// files (the "Software"), to deal in the Software without
// restriction, including without limitation the rights to use,
// copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following
// conditions:

// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.

#import "BTLSelfRemovingHandler.h"

@implementation BTLSelfRemovingHandler

- (void)readData:(BTLSocketBuffer*)someData fromAddress:(BTLSocketAddress*)anAddress
{
	BTLSocketBuffer* newData = [someData copy];
	++readWriteCount;
	[newData unwind];
	[newData addData:"a" ofSize:1];
	[self finishedReadingData:newData fromAddress:anAddress];
	
	[self reoveSocketIfNeeded];
}

- (void)writeData:(BTLSocketBuffer*)someData toAddress:(BTLSocketAddress*)anAddress
{
	BTLSocketBuffer* newData = [someData copy];
	++readWriteCount;
	[newData unwind];
	[newData addData:"a" ofSize:1];
	[self finishedWritingData:newData toAddress:anAddress];
	
	[self reoveSocketIfNeeded];
}

- (void)connectionFailedToAddress:(BTLSocketAddress*)anAddress
{
	[self writeData:[BTLSocketBuffer new] toAddress:anAddress];
}

- (void)connectionInterruptedToAddress:(BTLSocketAddress*)anAddress
{
	[self writeData:[BTLSocketBuffer new] toAddress:anAddress];
}

- (void)connectionClosedByRemoteAddress:(BTLSocketAddress*)anAddress
{
	[self writeData:[BTLSocketBuffer new] toAddress:anAddress];
}

- (void)setMaxCount:(int)aCount
{
	maxCount = aCount;
}

- (void)reoveSocketIfNeeded
{
	if(readWriteCount >= maxCount){
		[self removeSelf];
	}
}

- (id)copyWithZone:(NSZone*)zone
{
	id ret = [[self class] new];
	[ret setMaxCount:maxCount];
	return ret;
}

@end
